import time
import os
from PIL import ImageGrab

# 使用PIL的ImageGrab实现截图
# 也可以使用selenium提供的截图
def screenshot(name, save="Screenshot"):
    os.makedirs(save,exist_ok=True)
    t = time.time()
    png = ImageGrab.grab()
    image_name = os.path.join(save, name)
    png.save('%s_%s.png' % (image_name, str(round(t * 1000))))  # 文件名后面加了个时间戳，避免重名


if __name__ == "__main__":
    screenshot("test")